400f6ca74cff33aaa3738c7588a15eded1331b49,bgp/bmp-mock/src/test/java/org/opendaylight/protocol/bmp/mock/BmpMockTest.java,BmpMockTest,testMain,#,61

Before Change


        //1 * Initiate message + 3 * PeerUp Notification + 9 * Route Monitoring message
        Mockito.verify(this.sessionListener, Mockito.timeout(TimeUnit.SECONDS.toMillis(10)).times(13)).onMessage(Mockito.any(Notification.class));

        serverChannel.close().sync();
    }

    @Test

After Change


            serverChannel = futureServer.channel();
            sessionUpWait = 10;
        } else {
            serverChannel = null;
            // wait longer for the reconnection attempt
            sessionUpWait = 40;
        }

        BmpMock.main(new String[]{"--remote_address", InetSocketAddressUtil.toHostAndPort(serverAddr).toString(), "--peers_count", "3", "--pre_policy_routes", "3"});

        Mockito.verify(this.sessionListener, Mockito.timeout(TimeUnit.SECONDS.toMillis(sessionUpWait))).onSessionUp(Mockito.any(BmpSession.class));
        //1 * Initiate message + 3 * PeerUp Notification + 9 * Route Monitoring message
        Mockito.verify(this.sessionListener, Mockito.timeout(TimeUnit.SECONDS.toMillis(10)).times(13)).onMessage(Mockito.any(Notification.class));

        if (serverChannel != null) {
            serverChannel.close().sync();
        }
    }